iT邦幫忙

1

[Python]透過OpenCV讀取RTSP訊號呈現在Tkinter並錄影

  • 分享至 

  • xImage
  •  

前言

原本採用web架構,但web對於RTSP支援程度較差(MJPG才是網頁常見的格式,原生支援度高),目前使用的作法是透過架NodeJS接收RTSP串流,再經由websocket傳輸到瀏覽器畫面上,但這樣多一轉換層,除了效能的疑慮,也會有一些系統不穩定的風險,假如websocket斷掉,影像就進不來了,而且同一時間要連的RTSP不只有一個,不確定websocket是否能承受大量的資訊傳輸,因此覺得還是client直接接收RTSP串流比較穩一點;原本的想法是在react裡面埋nodejs,透過VNC抓RTSP的dll元件來去達成目的,但工程浩大,也還不知道怎麼做,最後整個重寫,使用python tkinter來呈現RTSP攝影機的影像了。

安裝opencv for python套件

這篇的重點就是要用OpenCV來讀RTSP,當然套件是不能少的,先pip一下吧

> pip install opencv-python

接著讀取RTSP的影像串流,呈現在畫面上撥放它

URL我們先用國外測試的RTSP網址:rtsp://wowzaec2demo.streamlock.net/vod/mp4:BigBuckBunny_115k.mov,用VNC稍作測試可以播放後,就把它放進來opencv套件吧

from cv2 import cv2

video = cv2.VideoCapture('rtsp://wowzaec2demo.streamlock.net/vod/mp4:BigBuckBunny_115k.mov')

這樣子相當於已經準備好跟RTSP的連線了,再來我們使用read()來取每一格frame,那他回傳的是一個陣列,第一個回傳狀態,第二個才是frame,所以我們用(status,frame)去接他,並用while去接每一格frame,while跳出條件當然是status囉,也就是當RTSP串流播放完畢跳出迴圈,另外,我會習慣再加上手動停止的flag,所以看起來會是這樣...

(status, frame) = video.read()
while self.__active and status:
    (status, frame) = video.read()

取到frame後,會處理一個色彩空間轉換的步驟,把BGR轉換成我們常見的RGBA,讓他生成pixel陣列後,再透過PIL套件轉成Image物件

imgArray = cv2.cvtColor(frame, cv2.COLOR_BGR2RGBA)
image = Image.fromarray(imgArray)
photo = ImageTk.PhotoImage(image=image)

最後,把RTSP影像放在tkinter的Label widget做呈現囉

# 建立承載RTSP影像串流的容器物件
window = tk.Label(bg="black", image=photo, width=320, height=240)
# avoid garbage collection(避免資源被回收)(把圖片註冊到物件中,並自訂屬性,避免被回收)
window.rtspImage = photo
# 放置在畫面上
window.place(x=0, y=0, anchor='nw')

最後的最後,非常重要的一環,資源釋放,一定要切斷跟RTSP的連線,否則容易占用記憶體空間

video.release()

我到底看了什麼? 錄下來吧

到這邊,客戶又給了一個新需求,要再檢視攝影機RTSP串流的時候,同步錄影下來,還好我們有OpenCV大神加持,沒問題,馬上錄;使用VideoWriter()提供的寫入影像功能就可以達成目的了

record = cv2.VideoWriter([檔案路徑名稱], [影片編碼], [FPS], ([影片寬度], [影片高度]))

但是筆者在實做過程中,編碼如果跟來源不一樣,都沒有辦法成功錄進去檔案裡,於是再多做了一些處理,就是取得來源RTSP串流的編碼、FPS與寬高

# 取得來源RTSP影像的編碼
def getSourceFourcc(sourceVideo):
    code = sourceVideo.get(cv2.CAP_PROP_FOURCC)
    code = "".join([chr((int(code) >> 8 * i) & 0xFF) for i in range(4)])
    return code

recordForucc = cv2.VideoWriter_fourcc(*getSourceFourcc(video))
recordFPS = int(video.get(cv2.CAP_PROP_FPS))
recordWidth = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
recordHeight = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
# 建立錄影實體物件
record = cv2.VideoWriter('CameraRecord.avi', recordForucc, recordFPS, (recordWidth, recordHeight))

接著,在每次成功取得frame後,就write到檔案中,這個動作會根據設定的FPS來寫到影片檔案中

record.write(frame)  # 錄製影片到檔案

最後最後,很重要,要記得釋放,影片檔案才不會一直被占用住喔

record.release()

圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言